import torch

class LinearModel(torch.nn.Module):
    def __init__(self):
        super(LinearModel,self).__init__()
        self.linear1=torch.nn.Linear(4217,10)
        self.linear2 = torch.nn.Linear(10, 1)
    def forward(self,x):
        y_pred=self.linear1(x)
        y_pred = self.linear2(y_pred)
        return y_pred